package com.tanhua.autoconfig.template;

import com.baidu.aip.face.AipFace;
import org.json.JSONObject;
import org.springframework.beans.factory.annotation.Autowired;

import java.util.HashMap;

public class AipFaceTemplate {

    @Autowired
    private AipFace client;

    /**
     * 检测图片中是否包含人脸
     * true:包含
     * false:不包含
     */
    public boolean detect(String imageUrl) {
        //调用接口
        String imageType = "URL";

        HashMap<String,String> options = new HashMap<>();
        options.put("face_field","age");
        options.put("max_face_num","2");
        options.put("face_type","LIVE");
        options.put("liveness_control","LOW");

        //人脸检测
        JSONObject res = client.detect(imageUrl,imageType,options);
        System.out.println(res.toString(2));

        Integer error_code = (Integer) res.get("error_code");

        return error_code == 0;
    }
}
